import os

import nltk as nltk

from nltk.translate.meteor_score import meteor_score

from 翻译对比评分.读取xml文件特定信息 import meteor_score_fenci

if __name__ == '__main__':
    # 机器翻译
    candidate_translations = [
        'Snaptube 7 já está disponível!',
        'Velocidade de download extremamente rápida para todos os formatos.',
        'Pesquise o que quiser, baixe e jogue —— nunca foi tão fácil.',
        'O design da interface é limpo e moderno, o que encanta sua experiência de download.',
        'Suporte para reprodução e download de vídeos curtos.',
        'Shuffle, loop, gerenciamento de lista de reprodução etc.',
        'Letras legais de graça.',
        'Adequado para tela cheia perfeitamente',
        'Mudar velocidade para visualização',
        'Aplique agora',
        'Reproduzir como áudio',
        'Reproduzindo como áudio para visualização',
    ]

    # 人为翻译
    reference_translations = [
        ['O Snaptube 7 já está disponível! ', 'O Snaptube 7 já está disponível!'],
        ['Velocidade de download relâmpago para todos os formatos.',
         'Velocidade ultrarrápida de download para todos os formatos.'],
        ['Pesquise o que quiser, baixe e reproduza —— nunca foi tão fácil. ',
         'Pesquise o que quiser, baixe e reproduza —— nunca foi tão fácil.'],
        ['O design da interface é limpo e funcional, para uma experiência de download mais agradável. ',
         'O design da interface é limpo e fresco, encantando sua experiência de download. '],
        ['Suporte para reprodução e download de vídeos curtos',
         'Disponível para reprodução e download de vídeos curtos.'],
        ['Shuffle, loop, gerenciamento de lista de reprodução etc.',
         'Aleatório, repetir tudo, gerencie search_text lista de reprodução, etc.'],
        ['Letras legais de graça.', 'Letras incríveis de graça.'],
        ['Adequado para tela cheia perfeitamente', 'Perfeito para tela cheia'],
        ['Mudar velocidade para visualização', 'Troque search_text velocidade para pré-visualização'],
        ['Aplicar agora', 'Inscrever agora'],
        ['Reproduzir como áudio', 'Reproduzir como áudio'],
        ['Reproduzindo como áudio para visualização', 'Reproduzindo como áudio para pré-visualização'],
    ]
    nltk.download('wordnet')
    # 逐个计算Meteor分数
    meteor_score_fenci(candidate_translations, reference_translations)
